@echo off
chcp 65001 >nul
echo ========================================
echo   SmartTrader Pro 一键安装脚本
echo ========================================
echo.

REM 检查Python环境
echo [1/4] 检查Python环境...
python --version >nul 2>&1
if errorlevel 1 (
    echo ❌ 错误: 未安装Python或未添加到PATH
    echo 💡 请先安装Python 3.11+
    echo 💡 下载地址: https://www.python.org/downloads/
    echo 💡 安装时请勾选"Add Python to PATH"
    pause
    exit /b 1
)

for /f "tokens=2" %%i in ('python --version 2^>^&1') do set PYTHON_VERSION=%%i
echo ✅ Python版本: %PYTHON_VERSION%

REM 进入后端目录
cd backend

REM 创建虚拟环境
echo [2/4] 创建Python虚拟环境...
if not exist "venv" (
    python -m venv venv
    if errorlevel 1 (
        echo ❌ 虚拟环境创建失败
        pause
        exit /b 1
    )
    echo ✅ 虚拟环境创建成功
) else (
    echo ✅ 虚拟环境已存在
)

REM 激活虚拟环境
call venv\Scripts\activate

REM 安装依赖
echo [3/4] 安装Python依赖包...
echo 📥 正在安装依赖，请稍候...
pip install --upgrade pip
pip install -r requirements.txt
if errorlevel 1 (
    echo ❌ 依赖安装失败，尝试使用国内镜像...
    pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
    if errorlevel 1 (
        echo ❌ 依赖安装失败
        echo 💡 请检查网络连接
        pause
        exit /b 1
    )
)
echo ✅ 依赖安装完成

REM 初始化数据库
echo [4/4] 初始化本地数据库...
python local_setup.py
if errorlevel 1 (
    echo ❌ 数据库初始化失败
    pause
    exit /b 1
)

echo.
echo 🎉 安装完成！
echo.
echo 📝 下一步操作:
echo 1. 运行 start.bat 启动后端服务
echo 2. 访问 http://localhost:8000/docs 查看API文档
echo 3. 测试API: http://localhost:8000/api/v1/stocks/list
echo.
echo 💡 如需修改配置，请编辑 backend\.env 文件
echo.
pause